Using LASSO to select features